from PIL import Image
from utils import detect_image


name_classes = ["background", "aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair",
                "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"]

img_path = './VOCdevkit/VOC2007/JPEGImages/2007_009052.jpg'
image = Image.open(img_path)
r_image = detect_image(image, name_classes=name_classes, device=device)

r_image.show()
r_image.save('./predict.png')
